from 综合案例_包操作案例.api.login import login  # 导入函数

# 数据
test_data = [
    {"username": "17777429832", "password": "123456"},
    {"username": "1234", "password": "123456"}
]

for i in test_data:  # 使用 for 循环遍历 test_data 中的每一条测试用例
    username_1 = i.get('username')  # 通过get()方法获取用户名数据赋值给username_1
    print(f"username:{username_1}")
    username_2 = login(username_1)  # 调用前面导入的 login 函数，传入手机号 ，把返回结果保存在 username_2 中
    print(username_2)  # 打印出返回结果

    # 断言 主要是用来判断实际结果是否符合预期
    if username_1 == "17777429832":
        assert "账号正确" == username_2  # 如果输入的是 "17777429832"，期望返回 "账号正确"
    else:
        assert "账号错误" == username_2  # 如果实际返回值不等于期望值，assert 就会抛出 AssertionError，表示测试失败！
